from selenium import webdriver
from selenium_utils import OSConfig
import time  # 新增导入time模块

# 配置谷歌浏览器静默启动【后台启动】
chrome_options = webdriver.ChromeOptions()
# 初始化谷歌浏览器的配置【必须加上】
chrome_options.add_argument('--headless')
chrome_options.add_argument('--no-sandbox')
chrome_options.add_argument('--disable-gpu')
chrome_options.add_argument('--disable-dev-shm-usage')
driver = webdriver.Chrome(options=OSConfig.get_chrome_options())

driver.get("https://www.baidu.com/")

time.sleep(4)  # 使用time.sleep()实现延时 这样避免网页没加载完

driver.save_screenshot('screenshot2.png')
driver.quit()
print("截图成功。")
